基于生成对抗网络的图像去雾算法

doi:10.16451/j.cnki.issn1003-6059.202111003

Abstract
Figure/Table
References
Related Citation (4)

Download: PDF (5484 KB) HTML (1 KB)
Export: BibTeX | EndNote (RIS)

Abstract Compared with the image dehazing methods based on image enhancement or physical model, the current image dehazing methods based on deep learning improve the computational efficiency to a certain extent. Nevertheless, the problems of incomplete dehazing and color distortion still exist in complex scenes. Aiming at the different perceptions of human eyes on global and local features, an algorithm of image dehazing based on generative adversarial networks is proposed. Firstly, a multi-scale generator network is designed. The full-size image and the segmented image block are taken as the input to extract the global contour information and local detail information of the image. Then, a feature fusion module is constructed to fuse the global and local information, and the authenticity of the generated dehazing image is judged by the discriminant network. To make the generated dehazing image closer to the corresponding real haze-free image, a multivariate joint loss function is designed by combining the dark channel prior loss, the adversarial loss, the structural similarity loss and the smooth L1 loss to train the network. Experimental results show that the proposed algorithm is superior to some state-of-the-art dehazing algorithms.

Key words： Image Dehazing Generative Adversarial Networks(GAN) Multi-scale Structure Dark Channel Prior Multivariate Joint Loss

Received: 10 May 2021

ZTFLH:

TP 391

Fund:National Natural Science Foundation of China(No.61862030,62072218), Natural Science Foundation of Jiangxi Province(No.20192ACB20002,20192ACBL21008)

Corresponding Authors: YANG Yong, Ph.D., professor. His research interests include image processing and deep learning.

About author:: HUANG Shuying, Ph.D., associate professor. Her research interests include image processing and machine learning.
WANG Bin, master student. His research interests include deep learning, low-quality image enhancement and image dehazing.
LI Hongxia, master student. Her research interests include image dehazing and low-light image enhancement.
HU Wei, master student. His research interests include image dehazing and image enhancement.

	Service

	E-mail this article
	Add to my bookshelf
	Add to citation manager
	E-mail Alert
	RSS
	Articles by authors
	HUANG Shuying
	WANG Bin
	LI Hongxia
	YANG Yong
	HU Wei

Cite this article:

HUANG Shuying,WANG Bin,LI Hongxia等. Image Dehazing Based on Generative Adversarial Network[J]. , 2021, 34(11): 990-1003.

URL:

http://manu46.magtech.com.cn/Jweb_prai/EN/10.16451/j.cnki.issn1003-6059.202111003 OR http://manu46.magtech.com.cn/Jweb_prai/EN/Y2021/V34/I11/990

[1] WANG W C, YUAN X H. Recent Advances in Image Dehazing. IEEE/CAA Journal of Automatica Sinica, 2017, 4(3): 410-436.
[2] AL-SAMMARAIE M F. Contrast Enhancement of Roads Images with Foggy Scenes Based on Histogram Equalization // Proc of the 10th International Conference on Computer Science and Education. Washington, USA: IEEE, 2015: 95-101.
[3] 杨万挺,汪荣贵,方帅,等.滤波器可变的Retinex雾天图像增强算法.计算机辅助设计与图形学学报, 2010, 22(6): 965-971.
(YANG W T, WANG R G, FANG S, et al. Variable Filter Retinex Algorithm for Foggy Image Enhancement. Journal of Computer-Aided Design and Computer Graphics, 2010, 22(6): 965-971.)
[4] SEOW M J, ASARI V K. Ratio Rule and Homomorphic Filter for Enhancement of Digital Colour Image. Neurocomputing, 2006, 69(7/8/9): 954-958.
[5] JIA J F, YUE H. A Wavelet-Based Approach to Improve Foggy Image Clarity. IFAC Proceedings Volumes, 2014, 47(3): 930-935.
[6] TAN R T. Visibility in Bad Weather from a Single Image // Proc of the IEEE Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2008. DOI: 10.1109/CVPR.2008.4587643.
[7] HE K M, SUN J, TANG X O. Single Image Haze Removal Using Dark Channel Prior. IEEE Transactions on Pattern Analysis and Machine Intelligence, 2011, 33(12): 2341-2353.
[8] MENG G F, WANG Y, DUAN J Y, et al. Efficient Image Dehazing with Boundary Constraint and Contextual Regularization // Proc of the IEEE International Conference on Computer Vision. Washington, USA: IEEE, 2013: 617-624.
[9] ZHENG M Y, QI G Q, ZHU Z Q, et al. Image Dehazing by an Artificial Image Fusion Method Based on Adaptive Structure Decomposition. IEEE Sensors Journal, 2020, 20(14): 8062-8072.
[10] JU M Y, DING C, GUO Y J, et al. IDGCP: Image Dehazing Based on Gamma Correction Prior. IEEE Transactions on Image Proce-ssing, 2019, 29: 3104-3118.
[11] CAI B L, XU X M, JIA K, et al. DehazeNet: An End-to-End System for Single Image Haze Removal. IEEE Transactions on Image Processing, 2016, 25(11): 5187-5198.
[12] LI B Y, PENG X L, WANG Z Y, et al. AOD-Net: All-in-One Dehazing Network // Proc of the IEEE International Conference on Computer Vision. Washington, USA: IEEE, 2017: 4770-4778.
[13] CHEN D D, HE M M, FAN Q N, et al. Gated Context Aggregation Network for Image Dehazing and Deraining // Proc of the IEEE Winter Conference on Applications of Computer Vision. Washington, USA: IEEE, 2019: 1375-1383.
[14] LIU Z, XIAO B T, ALRABEIAH M, et al. Single Image Dehazing with a Generic Model-Agnostic Convolutional Neural Network. IEEE Signal Processing Letters, 2019, 26(6): 833-837.
[15] QU Y Y, CHEN Y Z, HUANG J Y, et al. Enhanced Pix2pix Dehazing Network // Proc of the IEEE Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2019: 8152-8160.
[16] ZHU H Y, XI P, CHANDRASEKHAR V, et al. DehazeGAN: When Image Dehazing Meets Differential Programming // Proc of the 27th International Joint Conference on Artificial Intelligence. New York, USA: ACM, 2018: 1234-1240.
[17] ENGIN D, GENC A, EKENEL H K. Cycle-Dehaze: Enhanced CycleGAN for Single Image Dehazing // Proc of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops. Washington, USA: IEEE, 2018: 838-846.
[18] YANG X T, XU Z, LUO J B. Towards Perceptual Image Dehazing by Physics-Based Disentanglement and Adversarial Training[C/OL]. [2021-05-04]. http://www.xyang35.umiacs.io/files/perceptual-image-dehazing.pdf.
[19] REN W Q, LIU S, ZHANG H, et al. Single Image Dehazing via Multi-scale Convolutional Neural Networks // Proc of the European Conference on Computer Vision. Berlin, Germany: Springer, 2016: 154-169.
[20] ZHANG J, TAO D C. FAMED-Net: A Fast and Accurate Multi-scale End-to-End Dehazing Network. IEEE Transactions on Image Processing, 2020, 29: 72-84.
[21] DONG H, PAN J S, XIANG L, et al. Multi-scale Boosted Deha-zing Network with Dense Feature Fusion // Proc of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington, USA: IEEE, 2020: 2154-2164.
[22] YEH C H, HUANG C H, KANG L W. Multi-scale Deep Residual Learning-Based Single Image Haze Removal via Image Decomposition. IEEE Transactions on Image Processing, 2020, 29: 3153-3167.
[23] LI B Y, REN W Q, FU D P, et al. Benchmarking Single-Image Dehazing and Beyond. IEEE Transactions on Image Processing, 2019, 28(1): 492-505.
[24] WANG Z, BOVIK A C, SHEIKH H R, et al. Image Quality Assessment: From Error Visibility to Structural Similarity. IEEE
Transactions on Image Processing, 2004, 13(4): 600-612.
[25] HUYNH-THU Q, GHANBARI M. Scope of Validity of PSNR in Image/Video Quality Assessment. Electronics Letters, 2008, 44(13): 800-801.
[26] SHARMA G, WU W C, DALAL E N. The CIEDE2000 Color-Difference Formula: Implementation Notes, Supplementary Test Data, and Mathematical Observations. Color Research and Application, 2005, 30(1): 21-30.